/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package servlet;

import hibernate.AlbumDAO;
import hibernate.ImageDAO;
import java.io.IOException;
import java.io.PrintWriter;
import javax.servlet.RequestDispatcher;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import jaxb.AlbumContentMarshaller;
import jaxb.AlbumMarshaller;
import org.hibernate.Session;
import photogallery.Album;
import photogallery.HibernateUtil;
import photogallery.Image;

/**
 *
 * @author ThuanPQ
 */
public class Controller extends HttpServlet {

    /**
     * Processes requests for both HTTP <code>GET</code> and <code>POST</code> methods.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    protected void processRequest(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        response.setContentType("text/html;charset=UTF-8");
        PrintWriter out = response.getWriter();
        String realPath = getServletContext().getRealPath("/") + "WEB-INF\\";
        try {
            String action = request.getParameter("action");
            String url = "";
            AlbumDAO abDAO = new AlbumDAO();
            Session session = null;

            if (action.equals("Create Album")) {
                String albumName = request.getParameter("txtAlbumName");
                Album[] albums = abDAO.findByAlbumName(albumName);
                Album album = null;
                if (albums == null || albums.length <= 0) {
                    album = abDAO.createAlbum(albumName);
                }
                url = "Controller?albumId=" + String.valueOf(album.getAlbumId()) + "&action=MarshallAlbumsDetail";
            } else if (action.equals("Search Album")) {
                String txtSearch = request.getParameter("txtSearch");
                Album[] result = abDAO.findByAlbumName(txtSearch);
                session = HibernateUtil.getSessionFactory().openSession();
                session.getTransaction().begin();
                request.setAttribute("INFO", result);
                url = "showalbum.jsp";
            } else if (action.equals("MarshallAlbumsDetail")) {
                AlbumMarshaller am = new AlbumMarshaller();
                am.createXML(realPath);
                url = "Controller?albumId=" + request.getParameter("albumId") + "&action=MarshallAlbumContent";
            } else if (action.equals("MarshallAlbumContent")) {
                Image[] images;
                Integer albumId = Integer.valueOf(request.getParameter("albumId"));
                AlbumContentMarshaller acm = new AlbumContentMarshaller();
                ImageDAO idao = new ImageDAO();
                images = idao.getAlbumImages(albumId);
                System.out.println("images.length: " + images.length);
                acm.createXML(albumId, images, realPath);
                url = "index.jsp";
            }

            RequestDispatcher rd = request.getRequestDispatcher(url);
            rd.forward(request, response);
            if (session != null) {
                session.getTransaction().commit();
            }
        } finally {
            out.close();
        }
    }

    // <editor-fold defaultstate="collapsed" desc="HttpServlet methods. Click on the + sign on the left to edit the code.">
    /**
     * Handles the HTTP <code>GET</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        processRequest(request, response);
    }

    /**
     * Handles the HTTP <code>POST</code> method.
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        processRequest(request, response);
    }

    /**
     * Returns a short description of the servlet.
     * @return a String containing servlet description
     */
    @Override
    public String getServletInfo() {
        return "Short description";
    }// </editor-fold>
}
